package com.xxxx.mail.receiver;

import com.rabbitmq.client.Channel;
import com.xxxx.server.pojo.AttdRecord;
import com.xxxx.server.pojo.MailConstants;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.amqp.support.AmqpHeaders;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.mail.MailProperties;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
import org.springframework.stereotype.Component;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;

import javax.mail.internet.MimeMessage;
import java.io.IOException;
import java.util.Date;

/**
 * 消息接收者
 *
 * @author: STFU
 * @create: 2021-04-08
 **/
@Component
public class MailReceiver {

    private static final Logger LOGGER = LoggerFactory.getLogger(MailReceiver.class);

    @Autowired
    private JavaMailSender javaMailSender;
    @Autowired
    private MailProperties mailProperties;
    @Autowired
    private TemplateEngine templateEngine;
    @Autowired
    private RedisTemplate redisTemplate;

    @RabbitListener(queues = MailConstants.MAIL_QUEUE_NAME)
    public void handler(Message message, Channel channel){
        AttdRecord leave = (AttdRecord) message.getPayload();
        MessageHeaders headers = message.getHeaders();
        //  消息序号(唯一)
        Long tag = (Long) headers.get(AmqpHeaders.DELIVERY_TAG);
        //  获取msgId
        String msgId = (String)headers.get("spring_returned_message_correlation");
        HashOperations hashOperations = redisTemplate.opsForHash();

        MimeMessage msg = javaMailSender.createMimeMessage();
        MimeMessageHelper helper = new MimeMessageHelper(msg);

        try {
            //  如果再次接收一样的消息
            if (hashOperations.entries("mail_log").containsKey(msgId)){
                LOGGER.error("消息已经被消费=========================={}",msgId);
                /**
                 * 手动确认消息
                 * tag：消息序号
                 * multiple：是否确认多条
                 */
                channel.basicAck(tag,false);
                return;
            }

            //发件人
            helper.setFrom(mailProperties.getUsername());
            //收件人
            helper.setTo(leave.getTeacher().getEmail());
            //主题
            helper.setSubject("您有一条请假申请待处理");
            //发送日期
            helper.setSentDate(new Date());
            //邮件内容
            Context context = new Context();
            context.setVariable("teacher_name", leave.getTeacher().getName());
            context.setVariable("student_name", leave.getStudentName());
            context.setVariable("record_time", leave.getRecordTime());
            context.setVariable("leave_time", leave.getLeaveTime());
            context.setVariable("course_name", leave.getCourse().getName());
            context.setVariable("reason", leave.getReason());
            String mail = templateEngine.process("mail", context);
            helper.setText(mail,true);
            //  发送邮件
            javaMailSender.send(msg);

            LOGGER.info("邮件发送成功");
            //  将消息id存入redis
            hashOperations.put("mail_log",msgId,"OK");
            //手动确认消息
            /**
             * 手动确认消息
             * tag：消息序号
             * multiple：是否确认多条
             */
            channel.basicAck(tag,false);
        } catch (Exception e) {
            //  手动确认消息
            try {
                /**
                 * 手动确认消息
                 * tag：消息序号
                 * multiple：是否确认多条
                 * requeue：是否退回到队列
                 */
                channel.basicNack(tag,false,true);
            } catch (IOException ioException) {
                LOGGER.error("邮件发送失败========================={}",e.getMessage());
            }
            LOGGER.error("邮件发送失败========================={}",e.getMessage());
        }

    }

}
